#!/bin/bash
#SBATCH --job-name="run_train_astro_rcnn"
#SBATCH --output="train_astro_rcnn.out"
#SBATCH --partition=gpu
#SBATCH --time=5:00:00
#SBATCH --ntasks-per-node=160
#SBATCH --mem-per-cpu=1200
#SBATCH --nodes=1
#SBATCH --sockets-per-node=2
#SBATCH --cores-per-socket=8
#SBATCH --threads-per-core=4
#SBATCH --export=ALL
#SBATCH --gres=gpu:v100:4
source activate wmcle-colinjb2
cd ~/astro_rcnn/
python3 astro_rcnn.py train trainingset,validationset --normalize zscore
